package com.demo.juc.pci;

import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * @Author: jxm
 * @Description: 阻塞队列完善生产者和消费者
 * @Date: 2022/4/4 21:41
 * @Version: 1.0
 */
public class BlockingQueueTest {

    public static void main(String[] args) {
        MyResource myResource = new MyResource(new ArrayBlockingQueue<String>(10));

        new Thread(()->{
            System.out.println(Thread.currentThread().getName()+"\t生产线程启动");
            try {
                myResource.myProd();
            } catch (Exception e) {
                e.printStackTrace();
            }
        },"Prod").start();

        new Thread(()->{
            System.out.println(Thread.currentThread().getName()+"\t消费线程启动");
            try {
                myResource.myConsumer();
                System.out.println("-------------");
            } catch (Exception e) {
                e.printStackTrace();
            }
        },"Consumer").start();

        try {
            TimeUnit.SECONDS.sleep(5);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }


        System.out.println("5停止线程--------，活动结束");

        try {
            myResource.stop();
        } catch (Exception e) {
            e.printStackTrace();
        }


    }

}

class MyResource{

    //默认开启，进行生产+消费
    private volatile boolean FLAG = true;
    //原子引用
    private AtomicInteger atomicInteger = new AtomicInteger();

    BlockingQueue<String> blockingQueue = null;

    public MyResource(BlockingQueue<String> blockingQueue){
        this.blockingQueue = blockingQueue;
        System.out.println(Thread.currentThread().getName()+"\t");
    }

    public void myProd() throws Exception{

        String data = null;
        boolean retValue;

        while(FLAG){
            data = atomicInteger.incrementAndGet() + "";
            retValue  = blockingQueue.offer(data,2L,TimeUnit.SECONDS);
            if(retValue){
                System.out.println(Thread.currentThread().getName()+"\t"+"插入队列"+data+"成功");
            }else{
                System.out.println(Thread.currentThread().getName()+"\t"+"插入队列"+data+"失败");
            }
            TimeUnit.SECONDS.sleep(1);
        }
        System.out.println(Thread.currentThread().getName()+"\t"+"生产动作结束，表示FLAG = false");

    }

    public void myConsumer() throws Exception{

        String result = null;

        while (FLAG){
            result = blockingQueue.poll(2L,TimeUnit.SECONDS);
            if(null == result || result.equalsIgnoreCase("")){
                FLAG = false;
                System.out.println(Thread.currentThread().getName()+"\t"+"消费超时，退出");
                System.out.println();
                return;
            }

            System.out.println(Thread.currentThread().getName()+"\t"+"消费"+result+"成功");
        }

    }

    //停止线程
    public void stop() throws Exception{
        this.FLAG = false;
    }
}

